FROM nvidia/cuda:8.0-cudnn5-devel-ubuntu16.04
MAINTAINER sanny  #作者
#EXPOSE -p 80:8088 #设置端口

RUN apt-get update && apt-get install -y --no-install-recommends \
        build-essential \
	tmux \
	vim \
        cmake \
        git \
        wget \
	openssh-clients* \
        libatlas-base-dev \
        libboost-all-dev \
        libgflags-dev \
        libgoogle-glog-dev \
        libhdf5-serial-dev \
        libleveldb-dev \
        liblmdb-dev \
        libopencv-dev \
        libprotobuf-dev \
        libsnappy-dev \
        protobuf-compiler \
        python-dev \
        python-numpy \
        python-pip \
        python-setuptools \
        python-scipy && \
    rm -rf /var/lib/apt/lists/* && \
	echo "root:root" | chpasswd #设置密码

# ENV CAFFE_ROOT=/opt/caffe
# WORKDIR $CAFFE_ROOT
# # FIXME: use ARG instead of ENV once DockerHub supports this
# # https://github.com/docker/hub-feedback/issues/460
# ENV CLONE_TAG=1.0
# 
# RUN git clone -b ${CLONE_TAG} --depth 1 https://github.com/BVLC/caffe.git . && \
#     pip install --upgrade pip && \
#     cd python && for req in $(cat requirements.txt) pydot; do pip install $req; done && cd .. && \
#     git clone https://github.com/NVIDIA/nccl.git && cd nccl && make -j install && cd .. && rm -rf nccl && \
#     mkdir build && cd build && \
#     cmake -DUSE_CUDNN=1 -DUSE_NCCL=1 .. && \
#     make -j"$(nproc)"
# 
# ENV PYCAFFE_ROOT $CAFFE_ROOT/python
# ENV PYTHONPATH $PYCAFFE_ROOT:$PYTHONPATH
# ENV PATH $CAFFE_ROOT/build/tools:$PYCAFFE_ROOT:$PATH
# RUN echo "$CAFFE_ROOT/build/lib" >> /etc/ld.so.conf.d/caffe.conf && ldconfig

WORKDIR /workspace
